package config

import (
    "fmt"
    "log"

    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

var DB *gorm.DB

func InitDatabase() {
    dsn := "root:root@tcp(127.0.0.1:3306)/my_db?charset=utf8mb4&parseTime=True&loc=Local"
    
    var err error
    DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        log.Fatal("数据库连接失败: ", err)
    }
    
    log.Println("✅ 数据库连接成功")
}

func GetDB() *gorm.DB {
    if DB == nil {
        log.Fatal("数据库未初始化，请先调用 InitDatabase()")
    }
    return DB
}